package Javacode_7;
import java.util.Comparator;
import java.util.TreeSet;

// 学生类
class Student {
    private String name;
    private int age;

    // 构造方法
    public Student(String name, int age) {
        this.name = name;
        this.age = age;
    }

    // 重写toString方法
    @Override
    public String toString() {
        return  name + "   " + age;
    }

    // 提供获取姓名和年龄的方法（供Comparator使用）
    public String getName() {
        return name;
    }

    public int getAge() {
        return age;
    }
}

// 测试类
public class TreeSetDemo {
    public static void main(String[] args) {
        // 创建TreeSet集合，传入Comparator匿名内部类实现自定义排序
        TreeSet<Student> students = new TreeSet<>(new Comparator<Student>() {
            @Override
            public int compare(Student o1, Student o2) {
                // 1. 按姓名长度从小到大排序
                int nameLengthCompare = o1.getName().length() - o2.getName().length();

                // 2. 姓名长度相同时，按姓名内容排序
                if (nameLengthCompare == 0) {
                    int nameCompare = o1.getName().compareTo(o2.getName());

                    // 3. 姓名完全相同时，按年龄排序（保证唯一性）
                    if (nameCompare == 0) {
                        return o1.getAge() - o2.getAge();
                    }
                    return nameCompare;
                }
                return nameLengthCompare;
            }
        });

        // 向集合中添加学生对象
        students.add(new Student("linqingxia", 27));
        students.add(new Student("zhangguorong", 29));
        students.add(new Student("wanglihong", 23));
        students.add(new Student("linqingxia", 27)); // 重复对象，会被去重
        students.add(new Student("liushishi", 22));
        students.add(new Student("wuqilong", 40));
        students.add(new Student("gaoyuanyuan", 22));
        students.add(new Student("linqingxia", 29)); // 姓名相同年龄不同，视为不同对象

        // 遍历集合（按姓名长度从小到大排序）
        System.out.println("TreeSet中的学生对象（按姓名长度升序）：");
        for (Student student : students) {
            System.out.println(student);
        }
    }
}

